from pydantic import BaseModel

class ChatMessage(BaseModel):
    role: str
    content: str

class ChatCompletionRequest(BaseModel):
    model: str
    messages: list[ChatMessage]
    max_tokens: int = 1000
    temperature: float = 0.7
    top_p: float = 1.0
    n: int = 1
    stream: bool = False
